## 打印文件夹中所有的图片类型及对应数量
import os

# 指定文件夹路径
folder_path = '/home/shuai/notebook/note_scripts/boat/images_copy'

# 允许的图片扩展名列表
image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp', '.svg']

# 用于存储图片类型计数
extension_counts = {ext: 0 for ext in image_extensions}

# 遍历文件夹中的文件
for filename in os.listdir(folder_path):
    if os.path.isfile(os.path.join(folder_path, filename)):
        # 获取文件扩展名
        _, ext = os.path.splitext(filename)
        # 如果是图片文件的扩展名之一，计数
        if ext.lower() in image_extensions:
            extension_counts[ext.lower()] += 1

# 打印每种图片类型的数量
print("每种图片类型的数量：")
total_images = 0
for ext, count in extension_counts.items():
    if count > 0:
        print(f"{ext}: {count}")
        total_images += count

# 打印总图片数量
print(f"\n总共有 {total_images} 张图片")
